关闭tensorflow运行时的警告信息

执行简单的矩阵相乘的程序:

import tensorflow as tf


m1 = tf.constant([[3,3]])
m2 = tf.constant([[2],[3]])

product = tf.matmul(m1,m2)

print(product)

sess = tf.Session()

result = sess.run(product)
print(result)

sess.close()

输出:

Tensor("MatMul:0", shape=(1, 1), dtype=int32)
2019-03-18 10:44:35.419435: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2019-03-18 10:44:35.797977: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties: 
name: GeForce GTX 1050 major: 6 minor: 1 memoryClockRate(GHz): 1.493
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.61GiB
2019-03-18 10:44:35.798454: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1511] Adding visible gpu devices: 0
2019-03-18 10:44:37.076586: I tensorflow/core/common_runtime/gpu/gpu_device.cc:982] Device interconnect StreamExecutor with strength 1 edge matrix:
2019-03-18 10:44:37.076836: I tensorflow/core/common_runtime/gpu/gpu_device.cc:988]      0 
2019-03-18 10:44:37.076991: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1001] 0:   N 
2019-03-18 10:44:37.077308: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 1353 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1)
[[15]]

为了避免输出上面的警告信息,在程序开头加入:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
#默认为0:输出所有log信息
#设置为1:进一步屏蔽INFO信息
#设置为2:进一步屏蔽WARNING信息
#设置为3:进一步屏蔽ERROR信息

此时再执行上述的程序,输出:

Tensor("MatMul:0", shape=(1, 1), dtype=int32)
[[15]]
posted @ 2019-03-18 10:49  youngliu91  阅读(6578)  评论(0编辑  收藏  举报